adding functionality

adding functionality