Tweak code styles