fix PEP8, added TODO