Adding a display function until whole game is working