fixing code styling