The firm has doubled the warranty on its new flagship smartphones following criticisms of the hardware.