Code for the CMS DAS b tagging short exercise