The code must be Python 2 and 3 compatible. Please keep using the conditional import and make sure the rest of the code works with Python 3.