Added file for testing pipeline and also added TODOs to the classifier.py file