Added a small example
Small modifications in comments