Corrected an error in kicked-rotor.jl