Taking a different approach

And I figured out why my output was always wrong, it's because I wasn't
using the function passed in map, I was using the one on the case class,
and that one was always the same :(

    I'm dumb