Added test code

test code. Runs good.