framework for data extraction + workflow for prediction added