Add reference for button