updated TODO
updated TODO
updated TODO
fixed minor UI bug