2nd problem from L2 done.