Wrote some integration tests!