Add missing stuff to make it actually build