Add velocity rescaling