Create weight.py