First draaft, adding digit recognition for kaggle