Python script for calculating the kinematic equations