Added control flow test