Add check button and remove when check is called