First solution of the assignment