PEP-8, Python Zen, refactor and reduce code (thx FedeG)