Task 4 is now functional

Had to insert extra state ...