added a square for displaying