Changed button code