modified code styling