Adding import of re module