Add flow for buttons