Fixed errors
Added output to main.py, fixed PEP8