FIX: Added line to setup button events