week 4 assignment 3 - fixes