Add tests and flesh out readme