made a few small design changes