Use greedy subgradient; some fixes to plotting