Remove pyplot import