I think we should write a inherited class from this for testing instead of adding them directly here.