Updated menu program with switch statement