Add main function