You could add a few unit tests for this class