Cleaning up documentation and setting up demo in ViewController