Adding rust code