Adding tests and responses for image states