Only plot lines if activations are available
Updated demo
Improved plotting for function approximator training