Improve regex for defmodule highlighting.