diff --git a/astvisitor.cpp b/astvisitor.cpp
index c597b7b..f2348ab 100644
--- a/astvisitor.cpp
+++ b/astvisitor.cpp
@@ -1,244 +1,248 @@
 #include <clang/AST/ASTContext.h>
 
 #include "astvisitor.h"
 
 bool SmokegenASTVisitor::VisitCXXRecordDecl(clang::CXXRecordDecl *D) {
     // We can't make bindings for things that don't have names.
     if (!D->getDeclName())
         return true;
 
     registerClass(D);
 
     return true;
 }
 
 bool SmokegenASTVisitor::VisitEnumDecl(clang::EnumDecl *D) {
     // We can't make bindings for things that don't have names.
     if (!D->getDeclName())
         return true;
 
     //generator.addEnum(D);
 
     return true;
 }
 
 bool SmokegenASTVisitor::VisitNamespaceDecl(clang::NamespaceDecl *D) {
     if (!D->getDeclName())
         return true;
 
     registerNamespace(D);
 
     return true;
 }
 
 bool SmokegenASTVisitor::VisitFunctionDecl(clang::FunctionDecl *D) {
     if (clang::isa<clang::CXXMethodDecl>(D)){
         return true;
     }
 
     if (!D->getDeclName())
         return true;
 
     //generator.addFunction(D);
 
     return true;
 }
 
 clang::QualType SmokegenASTVisitor::getReturnTypeForMethod(const clang::CXXMethodDecl* method) const {
     if (clang::isa<clang::CXXConstructorDecl>(method)) {
         return ci.getASTContext().getPointerType(clang::QualType(method->getParent()->getTypeForDecl(), 0));
     }
     else {
         return method->getReturnType();
     }
 }
 
 Access SmokegenASTVisitor::toAccess(clang::AccessSpecifier clangAccess) const {
     Access access;
     switch (clangAccess) {
         case clang::AS_public:
         case clang::AS_none:
             access = Access_public;
             break;
         case clang::AS_protected:
             access = Access_protected;
             break;
         case clang::AS_private:
             access = Access_private;
             break;
     }
     return access;
 }
 
+Parameter SmokegenASTVisitor::toParameter(const clang::ParmVarDecl* param) const {
+    Parameter parameter(
+        QString::fromStdString(param->getNameAsString()),
+        registerType(param->getType())
+    );
+
+    if (const clang::Expr* defaultArgExpr = param->getDefaultArg()) {
+        std::string defaultArgStr;
+        llvm::raw_string_ostream s(defaultArgStr);
+        defaultArgExpr->printPretty(s, nullptr, pp());
+        parameter.setDefaultValue(QString::fromStdString(s.str()));
+    }
+
+    return parameter;
+}
+
 Class* SmokegenASTVisitor::registerClass(const clang::CXXRecordDecl* clangClass) const {
     clangClass = clangClass->hasDefinition() ? clangClass->getDefinition() : clangClass->getCanonicalDecl();
 
     QString qualifiedName = QString::fromStdString(clangClass->getQualifiedNameAsString());
     if (classes.contains(qualifiedName) and not classes[qualifiedName].isForwardDecl()) {
         // We already have this class
         return &classes[qualifiedName];
     }
 
     QString name = QString::fromStdString(clangClass->getNameAsString());
     QString nspace;
     Class* parent = nullptr;
     if (const auto clangParent = clang::dyn_cast<clang::NamespaceDecl>(clangClass->getParent())) {
         parent = registerNamespace(clangParent);
         nspace = QString::fromStdString(clangParent->getQualifiedNameAsString());
     }
     else if (const auto clangParent = clang::dyn_cast<clang::CXXRecordDecl>(clangClass->getParent())) {
         parent = registerClass(clangParent);
     }
     Class::Kind kind;
     switch (clangClass->getTagKind()) {
         case clang::TTK_Class:
             kind = Class::Kind_Class;
             break;
         case clang::TTK_Struct:
             kind = Class::Kind_Struct;
             break;
         case clang::TTK_Union:
             kind = Class::Kind_Union;
             break;
         default:
             break;
     }
 
     bool isForward = !clangClass->hasDefinition();
 
     Class localClass(name, nspace, parent, kind, isForward);
     classes[qualifiedName] = localClass;
     Class* klass = &classes[qualifiedName];
 
     klass->setAccess(toAccess(clangClass->getAccess()));
 
     if (clangClass->getTypeForDecl()->isDependentType() || clang::isa<clang::ClassTemplateSpecializationDecl>(clangClass)) {
         klass->setIsTemplate(true);
     }
 
     if (!isForward && !clangClass->getTypeForDecl()->isDependentType()) {
         // Set base classes
         for (const clang::CXXBaseSpecifier& base : clangClass->bases()) {
             const clang::CXXRecordDecl* baseRecordDecl = base.getType()->getAsCXXRecordDecl();
 
             if (!baseRecordDecl) {
                 // Ignore template specializations
                 continue;
             }
 
             Class::BaseClassSpecifier baseClass = Class::BaseClassSpecifier {
                 &classes[QString::fromStdString(baseRecordDecl->getQualifiedNameAsString())],
                 toAccess(base.getAccessSpecifier()),
                 base.isVirtual()
             };
 
             klass->appendBaseClass(baseClass);
         }
 
         // Set methods
         for (const clang::CXXMethodDecl* method : clangClass->methods()) {
             Method newMethod = Method(
                 klass,
                 QString::fromStdString(method->getNameAsString()),
                 registerType(getReturnTypeForMethod(method)),
                 toAccess(method->getAccess())
             );
             if (const clang::CXXConstructorDecl* ctor = clang::dyn_cast<clang::CXXConstructorDecl>(method)) {
                 newMethod.setIsConstructor(true);
                 if (ctor->isExplicitSpecified()) {
                     newMethod.setFlag(Member::Explicit);
                 }
             }
             else if (clang::isa<clang::CXXDestructorDecl>(method)) {
                 newMethod.setIsDestructor(true);
             }
             newMethod.setIsConst(method->isConst());
             if (method->isVirtual()) {
                 newMethod.setFlag(Member::Virtual);
                 if (method->isPure()) {
                     newMethod.setFlag(Member::PureVirtual);
                 }
             }
             if (method->isStatic()) {
                 newMethod.setFlag(Member::Static);
             }
 
             for (const clang::ParmVarDecl* param : method->parameters()) {
-                Parameter parameter(
-                    QString::fromStdString(param->getNameAsString()),
-                    registerType(param->getType())
-                );
-
-                if (const clang::Expr* defaultArgExpr = param->getDefaultArg()) {
-                    std::string defaultArgStr;
-                    llvm::raw_string_ostream s(defaultArgStr);
-                    defaultArgExpr->printPretty(s, nullptr, pp());
-                    parameter.setDefaultValue(QString::fromStdString(s.str()));
-                }
-
-                newMethod.appendParameter(parameter);
+                newMethod.appendParameter(toParameter(param));
             }
 
             klass->appendMethod(newMethod);
         }
     }
     return klass;
 }
 
 Class* SmokegenASTVisitor::registerNamespace(const clang::NamespaceDecl* clangNamespace) const {
     clangNamespace = clangNamespace->getCanonicalDecl();
 
     QString qualifiedName = QString::fromStdString(clangNamespace->getQualifiedNameAsString());
     if (classes.contains(qualifiedName)) {
         // We already have this class
         return &classes[qualifiedName];
     }
 
     QString name = QString::fromStdString(clangNamespace->getNameAsString());
     QString nspace;
     Class* parent = nullptr;
     if (const auto clangParent = clang::dyn_cast<clang::NamespaceDecl>(clangNamespace->getParent())) {
         parent = registerNamespace(clangParent);
         nspace = QString::fromStdString(clangParent->getQualifiedNameAsString());
     }
     classes[qualifiedName] = Class(name, nspace, parent, Class::Kind_Class, false);
     classes[qualifiedName].setIsNameSpace(true);
     return &classes[qualifiedName];
 }
 
 Type* SmokegenASTVisitor::registerType(clang::QualType clangType) const {
     clang::QualType orig = clang::QualType(clangType);
 
     Type type;
 
     if (clangType->isReferenceType()) {
         type.setIsRef(true);
 
         clangType = clangType->getPointeeType();
     }
     while (clangType->isPointerType()) {
         type.setIsConstPointer(type.pointerDepth(), clangType.isConstQualified());
         type.setPointerDepth(type.pointerDepth() + 1);
 
         clangType = clangType->getPointeeType();
     }
 
     type.setIsConst(clangType.isConstQualified());
     type.setIsVolatile(clangType.isVolatileQualified());
 
     type.setName(QString::fromStdString(clangType.getAsString(pp())));
     type.setIsIntegral(clangType->isBuiltinType());
 
     if (const clang::CXXRecordDecl* clangClass = clangType->getAsCXXRecordDecl()) {
         type.setClass(registerClass(clangClass));
 
         const auto templateSpecializationDecl = clang::dyn_cast<clang::ClassTemplateSpecializationDecl>(clangClass);
         if (templateSpecializationDecl) {
             const auto & args = templateSpecializationDecl->getTemplateArgs();
             for (int i=0; i < args.size(); ++i) {
                 type.appendTemplateArgument(*registerType(args[i].getAsType()));
             }
         }
     }
     return Type::registerType(type);
 }
diff --git a/astvisitor.h b/astvisitor.h
index b574bf6..1315531 100644
--- a/astvisitor.h
+++ b/astvisitor.h
@@ -1,33 +1,34 @@
 #ifndef SMOKEGEN_ASTVISITOR
 #define SMOKEGEN_ASTVISITOR
 
 #include <clang/AST/RecursiveASTVisitor.h>
 #include <clang/Frontend/CompilerInstance.h>
 #include <clang/Sema/Sema.h>
 
 #include "type.h"
 
 class SmokegenASTVisitor : public clang::RecursiveASTVisitor<SmokegenASTVisitor> {
 public:
     SmokegenASTVisitor(clang::CompilerInstance &ci) : ci(ci) {}
 
     bool VisitCXXRecordDecl(clang::CXXRecordDecl *D);
     bool VisitEnumDecl(clang::EnumDecl *D);
     bool VisitNamespaceDecl(clang::NamespaceDecl *D);
     bool VisitFunctionDecl(clang::FunctionDecl *D);
 
 private:
     clang::PrintingPolicy pp() const { return ci.getSema().getPrintingPolicy(); }
 
     clang::QualType getReturnTypeForMethod(const clang::CXXMethodDecl* method) const;
 
     Class* registerClass(const clang::CXXRecordDecl* clangClass) const;
     Class* registerNamespace(const clang::NamespaceDecl* clangNamespace) const;
     Type* registerType(const clang::QualType clangType) const;
 
     Access toAccess(clang::AccessSpecifier clangAccess) const;
+    Parameter toParameter(const clang::ParmVarDecl* param) const;
 
     clang::CompilerInstance &ci;
 };
 
 #endif