Made code button better!