small design fixes