Added test code for import