modify the main function, add uniform