Working on the assignment