Arduino Cybot Light Follower[Test Code 1]

//http://www.youtube.com/watch?v=gqFn7G5TBcE

//Motor control pins

int rightfor = 11;
int rightback = 10;
int leftfor = 9;
int leftback = 6;

//Variables

int diff = 0;
int correctionspeed = 0;

//Prototypes
int correction(int diff);


void setup(){

  pinMode(A0, INPUT);
  pinMode(A1, INPUT);
  pinMode(rightfor, OUTPUT);
  pinMode(rightback, OUTPUT);
  pinMode(leftfor, OUTPUT);
  pinMode(leftback, OUTPUT);

}

void loop(){

  if( analogRead(A0) > 10 || analogRead(A1) > 10){
  while(diff < 10 ){

  analogWrite(rightfor, 225);
  analogWrite(rightback, 0);
  analogWrite(leftfor, 225);
  analogWrite(leftback, 0);

  diff = analogRead(A0) - analogRead(A1);
  if(diff < 0) diff = (-1) * diff;

  }

  correction(diff);

  if ( analogRead(A0) > analogRead(A1)){

  analogWrite(rightfor, 0);
  analogWrite(rightback, 0);
  analogWrite(leftfor, correctionspeed);
  analogWrite(leftback, 0);
  delay(100);

  }
  else if ( analogRead(A1) > analogRead(A0)){

    analogWrite(rightfor, correctionspeed);
    analogWrite(rightback, 0);
    analogWrite(leftfor, 0);
    analogWrite(leftback, 0);
    delay(100);

  }
  }
  else {

    analogWrite(rightfor, 0);
    analogWrite(rightback, 0);
    analogWrite(leftfor, 0);
    analogWrite(leftback, 0);

  }

}

int correction(int diff){

  correctionspeed = (diff * (225/400)) + 127;

}