Adds return to menu functionality