Added more code to the menu stuff