further changes based on bruce's feedback

Signed-off-by: Shwetank Dixit <emailshwetank@gmail.com>