Updated module with example hook implementation