Add missing date logic