Updated integration test for challenge