Fixed up tests, added images for tests