clean up. updated docs