getting closer to the class design