Fix second argument now being options

* Second argument is an options object instead of just a color function