Fixed code styling and alignment