Fixed code styling