small tweaks to design