Adding button check.