Update main.py

Changed to be object oriented. Added a reset option. May not run.