fixing some code alignment