Small fixes to assignment.