I think this works?
need to add tests!