Add comments to rk4