improve code no real changes