Added junit tests for the classes.