fixing little things in assign4