Using weight when training