updated assignment-6