Added friction to axial velocity class