minor charnge to the main method