I need to change this for class