adding TODO to the comments