Updating for review comments