Add import statements back in java file