feat: small design improvements