minor changes and added docs