Cleaned up the code a lot. Still need to add catch trials and some more variability. But main question is down.