made some changes for the new design