add solution for assignment.