Flax balance tweaking and bugfixes