Python Notebook Manipulate Plot