Change code to use buttons not background