Assignment 3.3: Made some improvment in the code to the menu