Add main function call back in