Made code a little bit more PEP8-styled.