Extending the class instead