Add clusters legend and make canvas size bigger

Fix some errors in grid_test