Change class names in prototype.py to be more clear.