tweak the code