Updated documentation and README with set up instructions. Minor code clean up.