fix code styling