Modifications suggested

Signed-off-by: vinish <vinish@zhservices.com>