changes to the main method