Add date checks - Add restore function