Edited menu and added more options