Minor fixes to comments

- Fixed comments that had not been updated after refactoring