improved tests and minor changes to the readme