adding return in main