remove pyplot import