Add malloc null check