modify stub to add functionality