Add some tests
Add todo