Added compression for HDF5 file
Minor improvements on plotting