added predict method to access predictions