changes to main method