Added header function in layout class