Add some rough tests and benchmarks