Add new tests for images