Update README
Fix tests