Adding the main function