modify py-challenge