Nit-pick changes

Signed-off-by: Chintan Shah <cgintan1998@gmail.com>
Fix more errors

Signed-off-by: Chintan Shah <cgintan1998@gmail.com>