Working on design of other classes