Rename StageClass to StageClass.py