Floppy Drivercode

int i = 0;
boolean dir = true;
boolean st = false;
int wait = 4;

void setup() {
  pinMode(2, OUTPUT);  //Step
  pinMode(3, OUTPUT);  //Direction
  digitalWrite(2, HIGH);
  digitalWrite(3, HIGH);
}

void loop() {
  if(i >= 156) {
    dir = false;
    digitalWrite(3, LOW);
  } else if(i <=0) {
    dir = true;
    digitalWrite(3, HIGH);
  }

  if(dir) {
    i++;
  } else {
    i--;

  }
  if(st) {
    digitalWrite(2, LOW);
    st = false;
  } else {
    digitalWrite(2, HIGH);
    st = true;
  }

  delay(wait);
}