Fix the critical bug with weights
Bump to version 0.2.2
Merge branch 'release-0.2.2' into develop