adding tests to existing classes, moving to tdd