edit files to add functionality