Add set_display function