Fix import error and added explicitly six as dependency