closes #40, make plotting pretty