minor fixes, updated TODO