Tests were revisited. Added image generation and testing.