new design for problem one