Add malloc check