added small design changes