Wrote additional code for buttons and checkbox