In-situ heating and cooling stages