code clean up
updated docs