update TODO
remove unnecessary loop