Fix print msg. Add TODO unittest