Updated code styling