Tweak code format