predict method now working