Added inheritance to class