Plot Complex Functions In Python With Return