Plot A Function In Python Using Return