Add ui tests