i'd recommend pulling this logic out into a module and either extending that module here or allowing the client to choose where they want to extend it. i can show you if that doesnt make sense