Add some missing imports