Add malloc calls