Add documentation; slight cleanup