More support for PE flows, cleanup & improved typing #125

Job	Run time
build	7s
integration-test (3.11) / integration_test	3m 50s
integration-test (3.10) / integration_test	3m 55s
integration-test (3.13) / integration_test	3m 52s
integration-test (3.12) / integration_test	4m 16s
	16m 0s

Provide feedback