playing around with classes and unit testing