Add colored button functionality