Addressed peer review comments