Added some missing use statements