Added magnetization feature to the transformer model