Python code for making videos of evolution