Re-wrote dropout section