fix indentation in FFNet and return of train function should be difference, not output