Updated UART code