Cleanup

- made PEP8 compatible
- `colors == None` becomes `colors is None`
- replaced undefined variable `i` with `0`