Adding UI tests