Small modifications to reduce