Adding in menu functionality