Your best bet is to save a copy of all the original scripts. you might also look into a source control software, like sourcetree, so it's easier to manage changes.

In most cases you are adding to the existing scripts, Ive left the method names in so you know where they go.

As UFE is not open source, I can't post the full classes, so Ive tried to only post changes.