Add basic integration tests