Refactored a little and produced output for classifier
Trying to refactor code and produce desired output for Ryan