problem 1-4 finished!