Add menu functionality