just changed the arguments of the main