Added integration tests + code to pass them