Added more tests, fixed a typo in the readme