instead of 0, 1, 2, can you make the keys 'all', 'imported', 'not_imported' ?