Adding functionality as class from codex