working on python model