add data and machine learning implmentation