Integration tests working, adding new integration tests