modifed the unit tests for simluation
formal problem statement