Adding malloc override support