Optimize storing the tetris map.

Now just store the color of the shape in the map