Fix weighting
Merge pull request #8 from VividCortex/fix-weighting

Fix weighting