Slight docs/cleaning