Add malloc and free define