Added the code for the clicklistener, method for marking buttons