Add basic tests and update readme