Preserve comments and formatting while remove function from import.