Class extension

Class extension