Small changes to home work