add return in main function