SAP Work Manager 6.3 - Adding WBS Element to Time Sheet Screens

We are on SAP WM 6.3. We want to add WBS Element to the Time Sheet Screens.

On the Agentry, CATSRecord Object, I see the WbsElement Property, but the CATSRecord.java does not have the WbsElement Property.

I started with creating a ZCATSRecord Object and adding the WbsElemenet Property in there.

TimesheetAddBAPI.java and TimeSheetUpdateBapi.java are using the CATSRecord.

Should I extend TimesheetAddBAPI.java & TimesheetUpdateBAPI.java, Or make a copy of those classes and update the code to use ZCATSRecord Object instead of CATSRecord Object?

Thanks,

Gopi.