Corrections made as per the review comments #reviewthis @nikhiljain12

Signed-off-by: ShubhraJain <00sankalp@gmail.com>