Extending the class seems better to me