Removed unneeded import and tidied up comment.