more little tweaks to improve the UI