Working on the third assignment