added probelm 4