Fixing code styling