fixed minor error in main