Write more tests
Update README
Refactor specs