Added getButton method