Added layout for tests