correcting an error sending the last commit - missed some classes in the
program design