You can do either depending on the look you want to achieve. For a color pop after styling, apply when look is complete. For a more lived-in look, apply product and then heat style