Updated interface for handling reference image