Adding imports without changing the code ?