Update of visitor.py