Updated Working Example
Minor changes