Adds function to test btn state