Working on more tests and the readme