added missing date import

When I tried to run I was getting errors complaining about DateTime not being defined. Adding this import fixed this for my machine.