Cameron Diaz: It's been 'heaven' being in lockdown with baby