Wrote initial k.cole function from Ari's script. Added timeStep scalar in line 19