Updated to new scaler interface rather than assuming scaling to a normal distribution
Changed train to have inputs and outptus be public