Completed updates to policy key UI