Made an edit and closes issue #2