make changes based on number of tx antenna

Signed-off-by: Chun-Yeow Yeoh <yeohchunyeow@gmail.com>