Added test code