More work towards initial thermal ensemble