updates assignment2 with actual code