getting the modified class passed in correctly