Keithley 2601A IV pulse sweep and measurement

I am using Keithley 2601A and feeding TSP files directly to the machine through GPIB connection. 

I want to do an IV pulse sweep where I output voltage and measure current. There is an already existing function in the machine "ConfigPulseVMeasureISweepLin()" and "InitiatePulseTest()" that does almost exactly what I want. The problem is that allowed total time is limited. I want to be able to run longer measurements than allowed. 

I couldn't find code version of these functions so that I can do the changes I want to make. Is there a way to overwrite these functions such that I can run longer pulses? Or is there an code version of these function somewhere?

Thanks in advance for any kind of help.

Do you have the Test Script Builder application?
If yes, at the TSP> prompt in the instrument console window you can have the full code listing of the factory KIPulse script printed out.  You can then copy and paste it to a tsp file for editing.
You'll need to load the edited version to your own script/revised function names.

At the TSP> prompt type this:
 
​​​​​​​TSP>script.factory.scripts.KIPulse.list()
 

Alternately, there are example scripts distributed with Test Script Builder or we have them posted to our Github site.
2600A or 2600B Pulsed Sweep Examples

Using .list() from instrument console: