Button functioning cleared up