Add function for inputing a test survey code automatically