Changed a few things so that it works. And it does!