0.4 - tweak the code