I'd guess this is wrong, since it's too late for me to actually work if this is correct/what is the correct solution.