improved behavior during non zero momentum