cleanup
updated TODO