Adding shader loop and branch constructs