Added menu functionalilty