Update TODO
Update test