Added virial stepping