Make `plot` work with one complex argument as in MATLAB