Solution for Pro1.1 of Week2