Update rk.py

More commonly used format of RK4.