Clean and make code RuboCop compliant

Signed-off-by: Harsh Gupta <gupta.harsh96@gmail.com>