Added missing imports

Another split error still occurs though