Added tests for import feature