add return for main function