Changed code to follow python coding style