Fixed Don's initial class to meet the model I want.