Fix tests and update readme