update support for <begin_display> and <end_display>