Few minor changes, added documentation