Added code to geenrate predictions