Update classes with pass gen.py