Updated the menu to be collapsible, and to remember state