fix small design issues