Working on simulation assignment