Made display work correctly