adding return to main