Make plotting work for functions instead of just raw columns.