Add activity indicator code