Fix hardcoding and make mask and output line up.

[syncrostone]

Previously, there was a hard coded delay between the end of the stimulus and the beginning of the output, but the mask started having the network match the output when the stimulus turned off.

TODO: fix tests before merging, make a test.