Fix lighting calulations to be model space