fixed up cs3411 ass1