Adding functionality to module