Added heading to menu, and added functionality to button