Added ability to return to the selection menu and choose a different type of problem