small changes to design