Update README
Add unit tests for models