Added test code for demonstration purposes