Adding 2 missing imports