Progress on pyzhao, still doesn't work
Added Sean's weight transfer code