Added class to sample codebase

Forgot to add to last PR. This is causing tests to fail