Altered main.py to not require passing sys args

Now the script should be more easily run