Add file saving and draw_curves script