Adding the improvement steps