Improve step3