Add script to sort imports in Python modules