Fix code alignment/formatting