small changes on design