Added tests
Edits to readme