button code fixes