add TODO to classes