Added more optional flags and fixed docstring
Script to create a ddiscrete dot mesh rather than a filled discrete heat map
Corrected class labels