Model Railway Buffer Stop remix

I wanted to use this with my H0 scale train. I setup an end stop on both ends of the track and being controlled by an Arduino Uno board, it can sense when the end stops are hit and send the train in the opposite direction so it just goes back and forth on a straight track. I didn't want it to just hit the buffer, slamming it each time....so I tweaked it a bit so I could put springs on it so it would have some relief when hitting the end stop.

You can find the Arduino project info here: https://www.hackster.io/iceman198/arduino-railroad-ee9ac8

Video of it in action: https://youtu.be/nY3DaJmARTQ