Updating functions after changes made during testing