Added moduleForModel helper with test