Adding in tests, and updating the readme