clean up docs and improvements