Working on matplotlib-animation