Multi-antenna communication in a wireless network