Add some type imports