fixing the homework