clean up
fixed TODO
updates