Updated typo and imported pyplot