Fixed main menu, optimized a lil bit

Got rid of some unnecessary code, added a menu, and an unnecessary
button. Added a little bit of comments too. Comments are nice.