tidied code, removed redundant file main.py