Adding menu funtionality