Adding tests for decorating visitor