add simulation code back