write some test code
slove problem 3 with this code