Implement sum of squares task (Rust)