Set up custom layouts and button