Nearly finished with the button handling/menu logic now...