Documentation for classes and functions in the code would be really nice. Currently there is almost nothing.