Added menu functionality