Task four complete

Copied over code