Minor code styling