Corrected code alignment