Cleanup:
- PEP8 inconsistencies
- additional minor changes
Example:
- Visualized in 2D instead of 3D