Move integration tests into a seperate file