Better way to manage action buttons