New function added, will be modifying it based on the menu necessities.