remove TODO flag, add comments