Edited most button code