Add in layout attribute to allow for passing of layout class