add flow class and changes due to it