2- Adding the Flux model