Styling layout edit buttons. Figuring out how to make them toggleable