Adding small menu functionality