update py code