Added method to handle menu