Added an option to overwrite timeline_class