fit and predict added