fix README & improve tests