updated design to reflect code