Add testing files, fix saving, and images