I'd keep the codes as it is. This class is not used during simulations. just for testing and meshing.