Implemented styling suggestions from houndci

Signed-off-by: Ujjwal Thaakar <ujjwalthaakar@gmail.com>