minor fixes after submission of HW