(minor) tweak code format