Add working model for python