Added imports

Apparently Ben forgot to add some imports