Visitor Class and Unit Tests