Almost done with predict file, currently not using the wrapper