Added method to check button status
Added missing return