Adding menu functionality