Added heat storage part

Basic implementation