change data type of main function