Fix up build file properly, I think. Should be good now