Add getButton method