Minor update to todo