Added return to main function