A method of a user equipment (UE) in a wireless communication system is provided. The method comprises receiving, from a base station (BS), configuration information identifying a set of physical uplink control channel (PUCCH) resources, wherein each of the set of PUCCH resources is identified via an identifier (ID) and information associated with a transmit (Tx) beam; receiving, from the BS, a downlink control information (DCI) format to schedule a transmission over a physical uplink shared channel (PUSCH); identifying scheduling information for the transmission over the PUSCH based on the DCI format; configuring a Tx beam for the PUSCH based on the configuration information for the set of PUCCH resources; and transmitting, to the BS, data over the PUSCH based on the scheduling information using the Tx beam.