addressing feedback for assignment 3