Tweak code and pre styles