Use decorators instead of just classes