Small changes on proof-step function