Add retention management