add class information to display element

related to issue #45