Moved setup to after instantiation for Simulation, moved external code imports to inside class