fix

corrected minor error in main cpp