Update sim.py

added new 2 columns, hope it will be working