Added an integration test of sorts