Corrected code alignment issue