Add a draw grid function