Added method to enable button events