RNG: Improve python script for plotting RNG output