Added get_display_callback, this version has some bugs for key-input