Zhanghao Wu, UC Berkeley