Modified timeline.py file