add TODO
add comments