Add initial menu functionality