Add edit icon and update message in container proof of concept