Updated fix for units bug

Rather than passing the region object into the plot function with a
density object the region units are also stored in the density object
now.