Added controls on menu screen
Reworked menu :(