Adding support for controlling Puma.