changes to extend functionality
fixed missing reference which stopped build