Added some extra classes to generic entity display