- switched to google style docstrings
- removed dead code
- started to annotate types