[ViktorasM]

Neat and simple implementation. Consider docstrings for describing methods, this tends to integrate with IDEs a lot better than comments.