RK4 algorithm for initial value problem by python