Solving harmonic oscillator problem in Runge-Kutta method | Python