Add date handling