Added some arguments for specifying colours, grid, etc.