add test code

I modified for practice.