Add simple integration tests