Changed the import structure and moved the initialization code to run.py