Add user test flag implementation