Added reset code for buttons