add importance weighting to master branch