minor change main program