Add the ability to set custom params