Fixed colnames for model data

Improved handling of transformed variables