Small updates to getAssignment function in assignments.js