Removed print and plt