made plots better... more flexible input options