Minor fixes.
* Reorganise vega spec to put important stuff at top
* Add vega_axes to guess which axes to display
* First fix for constant prop