Added predict method