Chaned selenium tests to generated code