Trying to visualize optimization prolems with matplotlib