Fixed the class inheritance, forgot super.