Add if/else for link/image