updated python challenge with greg's feedback