Added wgt_model.py