address comments in pr