Add classification code

This merge request adds code for the classification task