Update weights.py
Merge pull request #369 from Kosvor/master

Update weights.py