Now, with stimulus!

Modified all the relevant functions to enable stimulus