Updated to use byte array