code formatting
added option to return to previous menu