Add macro for malloc and free