Add some missing imports in timeline.py
Use the proper timeline