diff --git a/budget/manage.py b/budget/manage.py
index 8bb44452..0a662840 100755
--- a/budget/manage.py
+++ b/budget/manage.py
@@ -5,6 +5,8 @@ from flask_script import Manager, Command
 from flask_migrate import Migrate, MigrateCommand
 from werkzeug.security import generate_password_hash
 
+from .run import app
+from .models import db
 
 
 class GeneratePasswordHash(Command):
@@ -20,5 +22,10 @@ manager = Manager(app)
 manager.add_command('db', MigrateCommand)
 manager.add_command('generate_password_hash', GeneratePasswordHash)   
 
-if __name__ == '__main__':
+
+def main():
     manager.run()
+
+
+if __name__ == '__main__':
+    main()