modify main function so it returns consistent data types