still not working. I might have to use a loop, instead of fancy matrix math