Remove unused elements; add TODO section