Add main function check