My changes to torflow