Minor: code clean-up

-align comments properly

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