Added more to machine learning part including images