Adds tests
Fixes tests, updates readme