Added week four problem