Update main.py

Big update. Everything works. made it more visually appealing. also added an erase button. Should still be able to shorten code.