Added classes for unit-tests for all classes. Also added a borrowed function that I need to rewrite for the WordCloudGenerator.