Deleted main.py and added load/test.py correctly