Changed test class to take in an image