2b 7 seg LED counter using Push Button

/*2b  7 seg LED counter using Push Button
  The  seven segment LED will count from 0 to 9 in a infinite loop
  Components Required
   1 Arduino Board
   1 seven segment LED + 2 Resistors (220 ohm)
   1 Push Button + 1 Resistor (10K ohm)
*/

int a = 2;  //For displaying segment "a"
int b = 3;  //For displaying segment "b"
int c = 4;  //For displaying segment "c"
int d = 5;  //For displaying segment "d"
int e = 6;  //For displaying segment "e"
int f = 7;  //For displaying segment "f"
int g = 8;  //For displaying segment "g"
int dp = 9;

char inputButtonState;
int i = 0;

void setup() {
  pinMode(a, OUTPUT);  //A
  pinMode(b, OUTPUT);  //B
  pinMode(c, OUTPUT);  //C
  pinMode(d, OUTPUT);  //D
  pinMode(e, OUTPUT);  //E
  pinMode(f, OUTPUT);  //F
  pinMode(g, OUTPUT);  //G
  pinMode(dp, OUTPUT);  //G
  pinMode(10, INPUT);
  pinMode(13, HIGH);
}

void displayDigit(int digit)
{
 //Conditions for displaying segment a
 if(digit!=1 && digit != 4)
 digitalWrite(a,LOW);

 //Conditions for displaying segment b
 if(digit != 5 && digit != 6)
 digitalWrite(b,LOW);

 //Conditions for displaying segment c
 if(digit !=2)
 digitalWrite(c,LOW);

 //Conditions for displaying segment d
 if(digit != 1 && digit !=4 && digit !=7)
 digitalWrite(d,LOW);

 //Conditions for displaying segment e
 if(digit == 2 || digit ==6 || digit == 8 || digit==0)
 digitalWrite(e,LOW);

 //Conditions for displaying segment f
 if(digit != 1 && digit !=2 && digit!=3 && digit !=7)
 digitalWrite(f,LOW);
 if (digit!=0 && digit!=1 && digit !=7)
 digitalWrite(g,LOW);
}

void turnOff()
{
  digitalWrite(a,HIGH);
  digitalWrite(b,HIGH);
  digitalWrite(c,HIGH);
  digitalWrite(d,HIGH);
  digitalWrite(e,HIGH);
  digitalWrite(f,HIGH);
  digitalWrite(g,HIGH);
}


void loop() {
  inputButtonState = digitalRead(10);

  if(i>=10)
  i=0;

  if(inputButtonState == HIGH)
  {
   digitalWrite(13, HIGH);
    displayDigit(i);
    i++;
    while(inputButtonState == HIGH)
  {
    inputButtonState = digitalRead(10);
  }
  }
  else
  {
    turnOff();
  }
}