Added counts for loops and assignments