- PEP-8 fixes.
 - Removed unused import.
 - More documentation.