Improved coloring function and added method to set graph output.
Added an __init__.py to the density_estimation path.
Moved some code around and added some more flexibility.
Improved plotting capabilities.