Add listener code