Modified menu

Added a menu function for easier use