diff --git a/codebase/models/shared/__pycache__/__init__.cpython-39.pyc b/codebase/models/shared/__pycache__/__init__.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..68705a286d6830777a3860ff23011f6d5a57d6f2
Binary files /dev/null and b/codebase/models/shared/__pycache__/__init__.cpython-39.pyc differ
diff --git a/codebase/models/shared/__pycache__/causal_masked_flow.cpython-39.pyc b/codebase/models/shared/__pycache__/causal_masked_flow.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..09a7a7002bf33f2404894e26c5fc127cdfdc3b1c
Binary files /dev/null and b/codebase/models/shared/__pycache__/causal_masked_flow.cpython-39.pyc differ
diff --git a/codebase/models/shared/__pycache__/mask.cpython-39.pyc b/codebase/models/shared/__pycache__/mask.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..85d1a8137628917f2041eec5a5e903835f09e88d
Binary files /dev/null and b/codebase/models/shared/__pycache__/mask.cpython-39.pyc differ
diff --git a/data/flow_noise/train/a_10_1_19_7.png b/data/flow_noise/train/a_10_1_19_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..ffc83c330aa7a078b3a77200d1494e982d3ef50c
Binary files /dev/null and b/data/flow_noise/train/a_10_1_19_7.png differ
diff --git a/data/flow_noise/train/a_10_1_28_11.png b/data/flow_noise/train/a_10_1_28_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..092c199f129d6f6e60e6492b174b637991a8a3c6
Binary files /dev/null and b/data/flow_noise/train/a_10_1_28_11.png differ
diff --git a/data/flow_noise/train/a_10_3_33_6.png b/data/flow_noise/train/a_10_3_33_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..dbe1961899e404f4af8c4cf8dc37f637d2ea13f8
Binary files /dev/null and b/data/flow_noise/train/a_10_3_33_6.png differ
diff --git a/data/flow_noise/train/a_11_1_28_12.png b/data/flow_noise/train/a_11_1_28_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..7a76b7482499c8a53f8c3c57ec7c3baea2dddcaa
Binary files /dev/null and b/data/flow_noise/train/a_11_1_28_12.png differ
diff --git a/data/flow_noise/train/a_11_1_28_9.png b/data/flow_noise/train/a_11_1_28_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..2d7dfe0abf319129a4965bb494a9038df2327110
Binary files /dev/null and b/data/flow_noise/train/a_11_1_28_9.png differ
diff --git a/data/flow_noise/train/a_11_1_30_12.png b/data/flow_noise/train/a_11_1_30_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..879dc2f2558736b50c5951854a577b95e5b0da28
Binary files /dev/null and b/data/flow_noise/train/a_11_1_30_12.png differ
diff --git a/data/flow_noise/train/a_11_2_31_7.png b/data/flow_noise/train/a_11_2_31_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..626648c292a599e177b01ce5193959e5f0e5ff4c
Binary files /dev/null and b/data/flow_noise/train/a_11_2_31_7.png differ
diff --git a/data/flow_noise/train/a_11_2_35_8.png b/data/flow_noise/train/a_11_2_35_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..a0660b05a903803a897ad4f12b9dea92c7889d2e
Binary files /dev/null and b/data/flow_noise/train/a_11_2_35_8.png differ
diff --git a/data/flow_noise/train/a_11_2_44_12.png b/data/flow_noise/train/a_11_2_44_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..a07d9f60efbab2e64563163933f3a02db0b0ca74
Binary files /dev/null and b/data/flow_noise/train/a_11_2_44_12.png differ
diff --git a/data/flow_noise/train/a_12_1_39_14.png b/data/flow_noise/train/a_12_1_39_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..4b3b3d07496bee13a500beae174d3913f6bd1f0f
Binary files /dev/null and b/data/flow_noise/train/a_12_1_39_14.png differ
diff --git a/data/flow_noise/train/a_12_2_38_11.png b/data/flow_noise/train/a_12_2_38_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..be0f5905c453754475937b2a9d65deaac230aa36
Binary files /dev/null and b/data/flow_noise/train/a_12_2_38_11.png differ
diff --git a/data/flow_noise/train/a_12_3_32_6.png b/data/flow_noise/train/a_12_3_32_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..855071753db7580dc164a0d070cd5d5a6fd4c4dd
Binary files /dev/null and b/data/flow_noise/train/a_12_3_32_6.png differ
diff --git a/data/flow_noise/train/a_12_3_35_7.png b/data/flow_noise/train/a_12_3_35_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..87ab8672f5be16c78ef2dca62ca9cd5677d49027
Binary files /dev/null and b/data/flow_noise/train/a_12_3_35_7.png differ
diff --git a/data/flow_noise/train/a_12_3_53_14.png b/data/flow_noise/train/a_12_3_53_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..4f7e550a0985e292601f1ebab924e72bcab21c4c
Binary files /dev/null and b/data/flow_noise/train/a_12_3_53_14.png differ
diff --git a/data/flow_noise/train/a_13_2_44_12.png b/data/flow_noise/train/a_13_2_44_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..0ca63b060117fed341a9ad57071c8915f7de564e
Binary files /dev/null and b/data/flow_noise/train/a_13_2_44_12.png differ
diff --git a/data/flow_noise/train/a_13_3_45_12.png b/data/flow_noise/train/a_13_3_45_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..5dabf99875871a8d5f646b6895051fb3f7f347c3
Binary files /dev/null and b/data/flow_noise/train/a_13_3_45_12.png differ
diff --git a/data/flow_noise/train/a_14_3_40_9.png b/data/flow_noise/train/a_14_3_40_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..9c752a351e2f8008f0891e5d7c0deff5c417be2a
Binary files /dev/null and b/data/flow_noise/train/a_14_3_40_9.png differ
diff --git a/data/flow_noise/train/a_14_3_45_12.png b/data/flow_noise/train/a_14_3_45_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..777d7474720115861069a3b36a337d8ae119d393
Binary files /dev/null and b/data/flow_noise/train/a_14_3_45_12.png differ
diff --git a/data/flow_noise/train/a_14_3_48_13.png b/data/flow_noise/train/a_14_3_48_13.png
new file mode 100644
index 0000000000000000000000000000000000000000..79b888eb6acaf8ac5ee8b21c7247afb21619784f
Binary files /dev/null and b/data/flow_noise/train/a_14_3_48_13.png differ
diff --git a/data/flow_noise/train/a_15_1_31_13.png b/data/flow_noise/train/a_15_1_31_13.png
new file mode 100644
index 0000000000000000000000000000000000000000..11225ff793f659f62f822a609d60883a30513ad6
Binary files /dev/null and b/data/flow_noise/train/a_15_1_31_13.png differ
diff --git a/data/flow_noise/train/a_15_2_29_7.png b/data/flow_noise/train/a_15_2_29_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..6c9d3e146e02cc8b6ab592dd0399c15c87fa5170
Binary files /dev/null and b/data/flow_noise/train/a_15_2_29_7.png differ
diff --git a/data/flow_noise/train/a_15_2_38_10.png b/data/flow_noise/train/a_15_2_38_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..f57a2335fcafe74f034b8ade37de6c580203c38d
Binary files /dev/null and b/data/flow_noise/train/a_15_2_38_10.png differ
diff --git a/data/flow_noise/train/a_15_3_44_10.png b/data/flow_noise/train/a_15_3_44_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..175ede592f00b22752e9478af9b87e439f46add7
Binary files /dev/null and b/data/flow_noise/train/a_15_3_44_10.png differ
diff --git a/data/flow_noise/train/a_16_1_23_7.png b/data/flow_noise/train/a_16_1_23_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..ea6f77e97d569a898a8923480e4ff10f0b49a7f6
Binary files /dev/null and b/data/flow_noise/train/a_16_1_23_7.png differ
diff --git a/data/flow_noise/train/a_16_1_32_12.png b/data/flow_noise/train/a_16_1_32_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..6a518cf858a451e58ea123861ccb41c03aa512cd
Binary files /dev/null and b/data/flow_noise/train/a_16_1_32_12.png differ
diff --git a/data/flow_noise/train/a_16_2_33_10.png b/data/flow_noise/train/a_16_2_33_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a22f8423ec6fdc78046b3e74e9642e64b979e90
Binary files /dev/null and b/data/flow_noise/train/a_16_2_33_10.png differ
diff --git a/data/flow_noise/train/a_16_2_38_12.png b/data/flow_noise/train/a_16_2_38_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..763628851ce7eb6524b965d3921e5a769028dbcd
Binary files /dev/null and b/data/flow_noise/train/a_16_2_38_12.png differ
diff --git a/data/flow_noise/train/a_16_3_43_11.png b/data/flow_noise/train/a_16_3_43_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..d4c1e619d09a9df952250372eb55994ad36caffb
Binary files /dev/null and b/data/flow_noise/train/a_16_3_43_11.png differ
diff --git a/data/flow_noise/train/a_16_3_44_10.png b/data/flow_noise/train/a_16_3_44_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..b36cdd0fa6201e80a3db11a0f6f8a66db615f5a7
Binary files /dev/null and b/data/flow_noise/train/a_16_3_44_10.png differ
diff --git a/data/flow_noise/train/a_17_2_39_13.png b/data/flow_noise/train/a_17_2_39_13.png
new file mode 100644
index 0000000000000000000000000000000000000000..7c989dfa60536cea8c37b08374fa2a5649f22dc3
Binary files /dev/null and b/data/flow_noise/train/a_17_2_39_13.png differ
diff --git a/data/flow_noise/train/a_18_1_21_7.png b/data/flow_noise/train/a_18_1_21_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..eb2d777268eb21fcaaee68a45adba63f05456138
Binary files /dev/null and b/data/flow_noise/train/a_18_1_21_7.png differ
diff --git a/data/flow_noise/train/a_18_2_31_8.png b/data/flow_noise/train/a_18_2_31_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..48722472b292f7f684176a3e0ea5b573012840d5
Binary files /dev/null and b/data/flow_noise/train/a_18_2_31_8.png differ
diff --git a/data/flow_noise/train/a_18_2_35_8.png b/data/flow_noise/train/a_18_2_35_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..734e9b730586e1b66392fa13e9798c226ffa3819
Binary files /dev/null and b/data/flow_noise/train/a_18_2_35_8.png differ
diff --git a/data/flow_noise/train/a_18_3_42_10.png b/data/flow_noise/train/a_18_3_42_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..952f259c8ba9294c15212fd8e089314f7803655b
Binary files /dev/null and b/data/flow_noise/train/a_18_3_42_10.png differ
diff --git a/data/flow_noise/train/a_19_3_33_7.png b/data/flow_noise/train/a_19_3_33_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..c44989279ab0d383753619c6cd9ca5a2a4f74b24
Binary files /dev/null and b/data/flow_noise/train/a_19_3_33_7.png differ
diff --git a/data/flow_noise/train/a_19_3_38_9.png b/data/flow_noise/train/a_19_3_38_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..defe2c0e3edf6b22de46b86e1eff3eff2a41685b
Binary files /dev/null and b/data/flow_noise/train/a_19_3_38_9.png differ
diff --git a/data/flow_noise/train/a_19_4_50_11.png b/data/flow_noise/train/a_19_4_50_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..0f140e27b15859bfb80359f1e856cfb1e49a9d9c
Binary files /dev/null and b/data/flow_noise/train/a_19_4_50_11.png differ
diff --git a/data/flow_noise/train/a_20_3_48_12.png b/data/flow_noise/train/a_20_3_48_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..c2d4de9232050ab72c5a2860c25fe30290f0fd45
Binary files /dev/null and b/data/flow_noise/train/a_20_3_48_12.png differ
diff --git a/data/flow_noise/train/a_20_4_53_12.png b/data/flow_noise/train/a_20_4_53_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..5c9f38abb2e899c071c3e7e2d645c01ca363690b
Binary files /dev/null and b/data/flow_noise/train/a_20_4_53_12.png differ
diff --git a/data/flow_noise/train/a_21_3_46_10.png b/data/flow_noise/train/a_21_3_46_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..5722c020f7ef2181cc9a5cfea4106ceb45108f2b
Binary files /dev/null and b/data/flow_noise/train/a_21_3_46_10.png differ
diff --git a/data/flow_noise/train/a_22_1_28_9.png b/data/flow_noise/train/a_22_1_28_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..004e02652ba6c42bac18a16c1ad45f8eb9b9d9f2
Binary files /dev/null and b/data/flow_noise/train/a_22_1_28_9.png differ
diff --git a/data/flow_noise/train/a_23_2_36_9.png b/data/flow_noise/train/a_23_2_36_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..76e4340a6a68a54c41aa8544fa157d747447fd27
Binary files /dev/null and b/data/flow_noise/train/a_23_2_36_9.png differ
diff --git a/data/flow_noise/train/a_23_2_40_11.png b/data/flow_noise/train/a_23_2_40_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..7cf5cf0d40a8e0050ed9a02d8f8812f5e5d1b1aa
Binary files /dev/null and b/data/flow_noise/train/a_23_2_40_11.png differ
diff --git a/data/flow_noise/train/a_23_3_41_9.png b/data/flow_noise/train/a_23_3_41_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..bf6e091e92f719ebb0abd1d9c3f432c9034a508b
Binary files /dev/null and b/data/flow_noise/train/a_23_3_41_9.png differ
diff --git a/data/flow_noise/train/a_23_3_47_12.png b/data/flow_noise/train/a_23_3_47_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..eba5ca322cdef8f4bfafedec9cb5bf7a1be87f76
Binary files /dev/null and b/data/flow_noise/train/a_23_3_47_12.png differ
diff --git a/data/flow_noise/train/a_24_1_30_11.png b/data/flow_noise/train/a_24_1_30_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..877fed9f2b268dbd1ce5b4154beebbb959cbc47f
Binary files /dev/null and b/data/flow_noise/train/a_24_1_30_11.png differ
diff --git a/data/flow_noise/train/a_24_3_36_7.png b/data/flow_noise/train/a_24_3_36_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..05d5e69537074ae485c511318873cc3919311f06
Binary files /dev/null and b/data/flow_noise/train/a_24_3_36_7.png differ
diff --git a/data/flow_noise/train/a_24_4_43_8.png b/data/flow_noise/train/a_24_4_43_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..6ef42902ca3c5ad0dea837030799ee28700facba
Binary files /dev/null and b/data/flow_noise/train/a_24_4_43_8.png differ
diff --git a/data/flow_noise/train/a_25_1_36_12.png b/data/flow_noise/train/a_25_1_36_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..f49760cd881492c644e4bdd355d4cf2169f4d936
Binary files /dev/null and b/data/flow_noise/train/a_25_1_36_12.png differ
diff --git a/data/flow_noise/train/a_25_3_49_13.png b/data/flow_noise/train/a_25_3_49_13.png
new file mode 100644
index 0000000000000000000000000000000000000000..dc50239e845b5dc6e44bddadea8968a431e6e39d
Binary files /dev/null and b/data/flow_noise/train/a_25_3_49_13.png differ
diff --git a/data/flow_noise/train/a_26_2_31_7.png b/data/flow_noise/train/a_26_2_31_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..b3817d56eb7f8de9bb1da726137e84a253838a66
Binary files /dev/null and b/data/flow_noise/train/a_26_2_31_7.png differ
diff --git a/data/flow_noise/train/a_26_2_32_7.png b/data/flow_noise/train/a_26_2_32_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..e8e67422eb2152d54d215532a1d1f3224ca19ecc
Binary files /dev/null and b/data/flow_noise/train/a_26_2_32_7.png differ
diff --git a/data/flow_noise/train/a_26_2_46_14.png b/data/flow_noise/train/a_26_2_46_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..63d4e8e43dd9d460822a09717b43a9ec36985532
Binary files /dev/null and b/data/flow_noise/train/a_26_2_46_14.png differ
diff --git a/data/flow_noise/train/a_26_3_35_7.png b/data/flow_noise/train/a_26_3_35_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..ec3d4a6196c2a4ff1698280f310b9c83f6e6c9ef
Binary files /dev/null and b/data/flow_noise/train/a_26_3_35_7.png differ
diff --git a/data/flow_noise/train/a_26_3_37_8.png b/data/flow_noise/train/a_26_3_37_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..4595e72730b58a412fd0acfcbc38dae692137709
Binary files /dev/null and b/data/flow_noise/train/a_26_3_37_8.png differ
diff --git a/data/flow_noise/train/a_26_3_46_11.png b/data/flow_noise/train/a_26_3_46_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d827cc9e1428dd2d23a6eed18ea52dc0f1d05
Binary files /dev/null and b/data/flow_noise/train/a_26_3_46_11.png differ
diff --git a/data/flow_noise/train/a_27_2_37_9.png b/data/flow_noise/train/a_27_2_37_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..b51c1260352436ed63f12973adca5900a491ef38
Binary files /dev/null and b/data/flow_noise/train/a_27_2_37_9.png differ
diff --git a/data/flow_noise/train/a_27_3_30_6.png b/data/flow_noise/train/a_27_3_30_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..62b34c1e03ada791dc79961a89c81c37e83aae7e
Binary files /dev/null and b/data/flow_noise/train/a_27_3_30_6.png differ
diff --git a/data/flow_noise/train/a_27_3_50_12.png b/data/flow_noise/train/a_27_3_50_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..425184286ab3394f3258c5cd112a29348e2faca7
Binary files /dev/null and b/data/flow_noise/train/a_27_3_50_12.png differ
diff --git a/data/flow_noise/train/a_27_3_53_14.png b/data/flow_noise/train/a_27_3_53_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..353fb5bcc243531e05136a47330b0a9171f71cc3
Binary files /dev/null and b/data/flow_noise/train/a_27_3_53_14.png differ
diff --git a/data/flow_noise/train/a_27_4_47_9.png b/data/flow_noise/train/a_27_4_47_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..4ff592c356902fd7df858f50b3b91eaf4a5297b3
Binary files /dev/null and b/data/flow_noise/train/a_27_4_47_9.png differ
diff --git a/data/flow_noise/train/a_27_4_53_11.png b/data/flow_noise/train/a_27_4_53_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f0663ecd489ea38d0153b1fde75fdb2df954203
Binary files /dev/null and b/data/flow_noise/train/a_27_4_53_11.png differ
diff --git a/data/flow_noise/train/a_28_1_30_9.png b/data/flow_noise/train/a_28_1_30_9.png
new file mode 100644
index 0000000000000000000000000000000000000000..67944f5ab15510862dfdd66f47b38cd4774887e6
Binary files /dev/null and b/data/flow_noise/train/a_28_1_30_9.png differ
diff --git a/data/flow_noise/train/a_28_2_29_8.png b/data/flow_noise/train/a_28_2_29_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..35767e2addf6b2726ae20c7758f24a5b00200172
Binary files /dev/null and b/data/flow_noise/train/a_28_2_29_8.png differ
diff --git a/data/flow_noise/train/a_28_4_35_6.png b/data/flow_noise/train/a_28_4_35_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..60654dc1fb97370233c7e30e84c45a8b85fca86c
Binary files /dev/null and b/data/flow_noise/train/a_28_4_35_6.png differ
diff --git a/data/flow_noise/train/a_28_4_44_8.png b/data/flow_noise/train/a_28_4_44_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..8fc9160808e239f203ef764d95d88e82e3e775ed
Binary files /dev/null and b/data/flow_noise/train/a_28_4_44_8.png differ
diff --git a/data/flow_noise/train/a_29_4_48_10.png b/data/flow_noise/train/a_29_4_48_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..39760d2287430d4ae0cd5a3704cdd9492ad168d9
Binary files /dev/null and b/data/flow_noise/train/a_29_4_48_10.png differ
diff --git a/data/flow_noise/train/a_31_2_34_10.png b/data/flow_noise/train/a_31_2_34_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..b74e67d6d17a27b102c0c148d30518d7ba0a2157
Binary files /dev/null and b/data/flow_noise/train/a_31_2_34_10.png differ
diff --git a/data/flow_noise/train/a_31_2_43_12.png b/data/flow_noise/train/a_31_2_43_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..984417b9e9a0a12999d1389786d9cbe1f896f830
Binary files /dev/null and b/data/flow_noise/train/a_31_2_43_12.png differ
diff --git a/data/flow_noise/train/a_31_5_52_10.png b/data/flow_noise/train/a_31_5_52_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..5b5a746813d1de4ff24a638361ee7ab4ac540dc4
Binary files /dev/null and b/data/flow_noise/train/a_31_5_52_10.png differ
diff --git a/data/flow_noise/train/a_32_4_37_7.png b/data/flow_noise/train/a_32_4_37_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..c1b19017d4eb5cc282bfaac483768c393ad6e9e6
Binary files /dev/null and b/data/flow_noise/train/a_32_4_37_7.png differ
diff --git a/data/flow_noise/train/a_32_4_39_7.png b/data/flow_noise/train/a_32_4_39_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..b45c5f06ca100f7fa81f3ca8135eb46de58390ea
Binary files /dev/null and b/data/flow_noise/train/a_32_4_39_7.png differ
diff --git a/data/flow_noise/train/a_33_2_27_6.png b/data/flow_noise/train/a_33_2_27_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..04b00de939a14d884e9a4e91c0071c489d4be732
Binary files /dev/null and b/data/flow_noise/train/a_33_2_27_6.png differ
diff --git a/data/flow_noise/train/a_33_3_38_8.png b/data/flow_noise/train/a_33_3_38_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..4def82a52ba0968566b4e708a19458353fc450b7
Binary files /dev/null and b/data/flow_noise/train/a_33_3_38_8.png differ
diff --git a/data/flow_noise/train/a_33_3_43_10.png b/data/flow_noise/train/a_33_3_43_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a45c532837ee2c690c5d05245ba187d85803c6a
Binary files /dev/null and b/data/flow_noise/train/a_33_3_43_10.png differ
diff --git a/data/flow_noise/train/a_34_3_49_11.png b/data/flow_noise/train/a_34_3_49_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..d1357c3df51b2bedd2a0e0d53642938d95b307cc
Binary files /dev/null and b/data/flow_noise/train/a_34_3_49_11.png differ
diff --git a/data/flow_noise/train/a_34_4_42_8.png b/data/flow_noise/train/a_34_4_42_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..9a78021515aa4ec57f148acb9d4fe8848d802a31
Binary files /dev/null and b/data/flow_noise/train/a_34_4_42_8.png differ
diff --git a/data/flow_noise/train/a_34_4_43_8.png b/data/flow_noise/train/a_34_4_43_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..19af8993e28bab407a48b4aae501f5d901e4176a
Binary files /dev/null and b/data/flow_noise/train/a_34_4_43_8.png differ
diff --git a/data/flow_noise/train/a_5_1_18_6.png b/data/flow_noise/train/a_5_1_18_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..eb80428bc589572247fe18e3d6b7bf866437dcf6
Binary files /dev/null and b/data/flow_noise/train/a_5_1_18_6.png differ
diff --git a/data/flow_noise/train/a_5_1_20_6.png b/data/flow_noise/train/a_5_1_20_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..ed551932e43080318e2f1f30ac1b30b4d6d0d807
Binary files /dev/null and b/data/flow_noise/train/a_5_1_20_6.png differ
diff --git a/data/flow_noise/train/a_6_1_24_10.png b/data/flow_noise/train/a_6_1_24_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..c4d6ac38d64cf46933a8ccadf3f7f7114a9a7250
Binary files /dev/null and b/data/flow_noise/train/a_6_1_24_10.png differ
diff --git a/data/flow_noise/train/a_6_2_37_12.png b/data/flow_noise/train/a_6_2_37_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..0b96fa403e1900568b51a59122d2106c20d3d62c
Binary files /dev/null and b/data/flow_noise/train/a_6_2_37_12.png differ
diff --git a/data/flow_noise/train/a_6_2_40_14.png b/data/flow_noise/train/a_6_2_40_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..1ac8e985052bf29444d53de713481c34ccdc96e1
Binary files /dev/null and b/data/flow_noise/train/a_6_2_40_14.png differ
diff --git a/data/flow_noise/train/a_6_2_45_14.png b/data/flow_noise/train/a_6_2_45_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..c3878e543a9efd868d08e9a4c7443efac471d743
Binary files /dev/null and b/data/flow_noise/train/a_6_2_45_14.png differ
diff --git a/data/flow_noise/train/a_7_1_26_7.png b/data/flow_noise/train/a_7_1_26_7.png
new file mode 100644
index 0000000000000000000000000000000000000000..0ba885d77aa0dfa467014fe9f3f39c1721d3a462
Binary files /dev/null and b/data/flow_noise/train/a_7_1_26_7.png differ
diff --git a/data/flow_noise/train/a_7_1_33_12.png b/data/flow_noise/train/a_7_1_33_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..f2d0eb4f10ad584895b2bbd970ae6231a7fe7995
Binary files /dev/null and b/data/flow_noise/train/a_7_1_33_12.png differ
diff --git a/data/flow_noise/train/a_7_2_25_6.png b/data/flow_noise/train/a_7_2_25_6.png
new file mode 100644
index 0000000000000000000000000000000000000000..d17d3ec0bf969e3bad728987bebb66fd0a8f8260
Binary files /dev/null and b/data/flow_noise/train/a_7_2_25_6.png differ
diff --git a/data/flow_noise/train/a_7_3_46_10.png b/data/flow_noise/train/a_7_3_46_10.png
new file mode 100644
index 0000000000000000000000000000000000000000..9c08c83046612ec28fc3cb2c6534855af17d7d6d
Binary files /dev/null and b/data/flow_noise/train/a_7_3_46_10.png differ
diff --git a/data/flow_noise/train/a_7_3_54_14.png b/data/flow_noise/train/a_7_3_54_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..1c6425cb86eb88685f8b8327b8ceca5fd4963d2e
Binary files /dev/null and b/data/flow_noise/train/a_7_3_54_14.png differ
diff --git a/data/flow_noise/train/a_8_2_40_12.png b/data/flow_noise/train/a_8_2_40_12.png
new file mode 100644
index 0000000000000000000000000000000000000000..ad486d0c9f7c2735339edd537f6c074201b76e6e
Binary files /dev/null and b/data/flow_noise/train/a_8_2_40_12.png differ
diff --git a/data/flow_noise/train/a_8_2_41_14.png b/data/flow_noise/train/a_8_2_41_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..79598381a3687e87a4e4fdf3aa3b5165b172f027
Binary files /dev/null and b/data/flow_noise/train/a_8_2_41_14.png differ
diff --git a/data/flow_noise/train/a_8_3_38_8.png b/data/flow_noise/train/a_8_3_38_8.png
new file mode 100644
index 0000000000000000000000000000000000000000..29813bba0b6b109a350bbe8533697ce9a2122a57
Binary files /dev/null and b/data/flow_noise/train/a_8_3_38_8.png differ
diff --git a/data/flow_noise/train/a_9_2_42_14.png b/data/flow_noise/train/a_9_2_42_14.png
new file mode 100644
index 0000000000000000000000000000000000000000..91115f840d90d0a4ff12f0307c44756c3afb23fc
Binary files /dev/null and b/data/flow_noise/train/a_9_2_42_14.png differ
diff --git a/data/flow_noise/train/a_9_3_48_11.png b/data/flow_noise/train/a_9_3_48_11.png
new file mode 100644
index 0000000000000000000000000000000000000000..95f6d49c971caa43f237d2521710b475c969c346
Binary files /dev/null and b/data/flow_noise/train/a_9_3_48_11.png differ
diff --git a/models/__pycache__/__init__.cpython-39.pyc b/models/__pycache__/__init__.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..bf41c4beb0632935f4ecaee85c9ef7a7decc884b
Binary files /dev/null and b/models/__pycache__/__init__.cpython-39.pyc differ
diff --git a/models/__pycache__/icm_vae.cpython-39.pyc b/models/__pycache__/icm_vae.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..62553a693c67a7658a88ab152bb1eb3f9aed5f38
Binary files /dev/null and b/models/__pycache__/icm_vae.cpython-39.pyc differ
diff --git a/models/__pycache__/prior.cpython-39.pyc b/models/__pycache__/prior.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..d1ac6f989d6ab5827208f620586b6068552128e2
Binary files /dev/null and b/models/__pycache__/prior.cpython-39.pyc differ