Homework small fixes