Added menu function to .h file