Added code for units in data driven problem author