Adding code to find the initialization block that precedes the loop