Moved the grid into its own class and draw the grid using pygame