Updated the inheritance that wasnt working