Changed test module to follow PEP8