Code review suggested changes

Signed-off-by: Lahiru Chandima <lahiruchandima@Lahirus-MacBook-Pro.local>