REF: Add %matplotlib magic