Added partial_fit and warm_start