Adding support for passing the controller response, closing #12