More documentation, small improvements