Kai Wang, UCLA