Code changes for review comments

Changed files for code formatting issue

Signed-off-by: Jay Patel <jaypa@cybage.com>