Added IRQ setup function