Ziqi added baseline in main function