Clean up documentation
change plotting