Add db argument to controller and listener
Make your code look good :)