I suck at python (hopefully fixed menu)