Changed display a little, still hacky fix this