diff --git a/examples/pinn_ssa.py b/examples/pinn_ssa.py
index 26a8115..8c837e9 100644
--- a/examples/pinn_ssa.py
+++ b/examples/pinn_ssa.py
@@ -9,11 +9,11 @@
 
 # General parameters
 hp = {}
-hp["epochs"] = 1000
+hp["epochs"] = 800000
 hp["learning_rate"] = 0.001
 hp["loss_function"] = "MSE"
-hp["save_path"] = "./Models/Helheim_test"
-hp["is_save"] = False
+hp["save_path"] = "./Models/Helheim_test/"
+hp["is_save"] = True
 hp["is_plot"] = True
 
 # NN
@@ -23,8 +23,8 @@
 hp["num_layers"] = 6
 
 # domain
-hp["shapefile"] = "./dataset/Helheim_Big.exp"
-hp["num_collocation_points"] = 5000
+hp["shapefile"] = "./dataset/fastflow_CF.exp"
+hp["num_collocation_points"] = 9000
 
 # physics
 SSA = {}
@@ -33,7 +33,7 @@
 
 # data
 issm = {}
-issm["data_size"] = {"u":1000, "v":1000, "s":1000, "H":1000, "C":None, "vel":1000}
+issm["data_size"] = {"u":4000, "v":4000, "s":4000, "H":4000, "C":None, "vel":4000}
 issm["data_path"] = "./dataset/Helheim_fastflow.mat"
 hp["data"] = {"ISSM":issm}