assignment 11 milestone fixes