Small design tweak