added some functionality to the classes