minor changes to final assignment