Changed demo to reflect changes in the pull request.
Added demo menu.