barostat part is revised