Update todo
Edit unit test