Add main function call