Make layout a class variable