using latest code - their added back predict method