Improve comments a bit.
Better documentation.