Add test main function for scheduler