Updated menu function