Added the main function