Added argumnet to skip prediction file generation