Tweak some code