Waterfall plot added to pyKdVLib