Add importance weights