Model trainer and image processor added