Add sample test code