add a solution for the assignment