made some changes to design