1st challenge as a class

added a single test