Add in the edit class, still need to port all the control classes to use mock objects.