Add rust mode stuff