Mass Spring Damper System Python