Code passing with rust changes