updated challenge portion