Set api attribute for Download class. Update imports and mock #7