working but not PEP8-ified