Added tests for simulation correctness