1 files changed, 9 insertions, 5 deletions
diff --git a/budget/manage.py b/budget/manage.py
index f717fed..0a66284 100755
--- a/budget/manage.py
+++ b/budget/manage.py
@@ -1,12 +1,12 @@
 #!/usr/bin/env python
 
+from getpass import getpass
 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
-from getpass import getpass
+from .run import app
+from .models import db
 
 
 class GeneratePasswordHash(Command):
@@ -20,8 +20,12 @@ migrate = Migrate(app, db)
 
 manager = Manager(app)
 manager.add_command('db', MigrateCommand)
-manager.add_command('generate_password_hash', GeneratePasswordHash)
+manager.add_command('generate_password_hash', GeneratePasswordHash)   
 
 
-if __name__ == '__main__':
+def main():
     manager.run()
+
+
+if __name__ == '__main__':
+    main()