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