Change design a little