work on milestone 2