little clean up and doc