With only a little bit of effort, the drawing code can be made a lot more pythonic. `== True` and `== False` are never acceptable, and for loops simply don't execute if the iterator is empty - no need to check.
