Add UI tests with PO