Fixd set_display and tested it