Now resets properly and so on, can just rerun the python program