Update readme & edit tests for #1