Modification on control button