Work on display function, but it's wrong