Update problem-1.rs