Adding in comments per peer review notes