Update week1_sol.scala