Fixed output to better suit transformer
Completed first fully working version of the transformer that can handle multiple models in the same file