Added data provenance functionality in modelling.py