Please add tests for this class.