Add flag-tests script