minor changes to make it all work.
Merge pull request #11 from jrygh00/view_branch

minor changes to make it all work.