Implement `map_mut` on the Option type