Wrap everything into a class