Writing UI tests in code