type the code

add code and test