Update main.py

opens a pygame window, need to add event handling to make it close when you click the X