fixed minor typo in main
new diff