Add test_image function