Smart oil treatment