updated models traincrane code