Python Plot Linear Function Definition