Added comments and an inheritance structure to the Dia.