Review comments added

thanks!