Impurity transport in tokamak plasmas containing momentum sources