README formatting
Add screenshot to docs
More unit tests