Streamfunctions.py fixed ledgend saving