Corrected code misalignments