Added tests for import functionality