Update three_link_pendulum.py