initial solution for the assignment