Add function to sort imports